from .loss import SuperVoxelLoss2D, SuperVoxelLoss3D